import numpy as np
import torch
import random
import os

SEED = 42

# 设置NumPy随机种子
np.random.seed(SEED)

# 设置PyTorch随机种子
torch.manual_seed(SEED)
if torch.cuda.is_available():
    torch.cuda.manual_seed_all(SEED)

# 设置Python标准库random模块的随机种子
random.seed(SEED)

# 设置环境变量以增加随机操作的确定性
os.environ['PYTHONHASHSEED'] = str(SEED)